package com.sheng.leetcode.year2023.month08.day01;

import org.junit.Test;

import java.util.Arrays;

/**
 * @author liusheng
 * @date 2023/08/01
 * <p>
 * 2681. 英雄的力量<p>
 * <p>
 * 给你一个下标从 0 开始的整数数组 nums ，它表示英雄的能力值。如果我们选出一部分英雄，这组英雄的 力量 定义为：<p>
 * i0 ，i1 ，... ik 表示这组英雄在数组中的下标。<p>
 * 那么这组英雄的力量为 max(nums[i0],nums[i1] ... nums[ik])^2 * min(nums[i0],nums[i1] ... nums[ik]) 。<p>
 * 请你返回所有可能的 非空 英雄组的 力量 之和。由于答案可能非常大，请你将结果对 10^9 + 7 取余。<p>
 * <p>
 * 示例 1：<p>
 * 输入：nums = [2,1,4]<p>
 * 输出：141<p>
 * 解释：<p>
 * 第 1 组：[2] 的力量为 2^2 * 2 = 8 。<p>
 * 第 2 组：[1] 的力量为 1^2 * 1 = 1 。<p>
 * 第 3 组：[4] 的力量为 4^2 * 4 = 64 。<p>
 * 第 4 组：[2,1] 的力量为 2^2 * 1 = 4 。<p>
 * 第 5 组：[2,4] 的力量为 4^2 * 2 = 32 。<p>
 * 第 6 组：[1,4] 的力量为 4^2 * 1 = 16 。<p>
 * 第 7 组：[2,1,4] 的力量为 4^2 * 1 = 16 。<p>
 * 所有英雄组的力量之和为 8 + 1 + 64 + 4 + 32 + 16 + 16 = 141 。<p>
 * <p>
 * 示例 2：<p>
 * 输入：nums = [1,1,1]<p>
 * 输出：7<p>
 * 解释：总共有 7 个英雄组，每一组的力量都是 1 。所以所有英雄组的力量之和为 7 。<p>
 * <p>
 * 提示：<p>
 * 1 <= nums.length <= 10^5<p>
 * 1 <= nums[i] <= 10^9<p>
 * <p>
 * 来源：力扣（LeetCode）<p>
 * 链接：<a href="https://leetcode.cn/problems/power-of-heroes">2681. 英雄的力量</a><p>
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。<p>
 */
public class LeetCode2681 {

    @Test
    public void test01() {
        int[] nums = {2, 1, 4};
//        int[] nums = {1, 1, 1};
        System.out.println(new Solution().sumOfPower(nums));
    }
}

class Solution {
    public int sumOfPower(int[] nums) {
        // 求所有数组的子数组，中最大值的平方和最小值之和的乘积之和
        Arrays.sort(nums);
        long p = 0, sum = 0;;
        for (int i = nums.length - 1; i >= 0; i--) {
            long num = nums[i];
            sum = (sum + (num * num % 1000000007) * nums[i]) % 1000000007;
            sum = (sum + num * p % 1000000007) % 1000000007;
            p = ((p * 2 + num * num % 1000000007) % 1000000007);
        }
        return (int) sum;
    }
}
